how to use python ones zeros full like array